package com.ruoyi.framework.auth.config;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.context.model.SaResponse;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.router.SaHttpMethod;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.CharsetUtil;
import com.isolar.comps.core.utils.SpringUtils;
import com.ruoyi.common.auth.handler.AllUrlHandler;
import com.ruoyi.framework.common.GlobalExceptionUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

/**
 * 权限安全配置
 *
 * @author Lion Li
 */

@Slf4j
//@AutoConfiguration
@Configuration
@EnableConfigurationProperties(SecurityProperties.class)
@RequiredArgsConstructor
public class SaTokenConfigure implements WebMvcConfigurer {

    private final SecurityProperties securityProperties;

    //private final  AdminInterceptor adminInterceptor;
    /**
     * 无需登录的接口地址集合
     */
    public static final String[] NO_LOGIN_PATH_ARR = {
            /* 主入口 */
            "/",
            "/login",
            "/logout",
            /* 静态资源 */
            "/webjars/**",
            /* 移动端静态资源 */
            "/assets/**",
            /* 认证相关 */
            "/auth/b/getPicCaptcha",
            "/auth/b/getPhoneValidCode",
            "/auth/b/doLogin",
            "/auth/b/doLoginByPhone",
            /* 系统基础配置 */
            "/dev/config/sysBaseList",
            /* 系统字典树 */
            "/dev/dict/tree",
            /* 文件下载 */
            "/static/*",
            "/dbapi/*",
            "/favicon.ico**",
            "/station.png",
            "ruoyi.png",
            "/html/**",
            "/css/**",
            "/docs/**",
            "/fonts/**",
            "/img/**",
            "/js/**",
            "/ajax/**",
            "/station/**",
            "/captcha/captchaImage**",
            /* 用户个人中心相关 */
            "/sys/userCenter/getPicCaptcha",
            "/sys/userCenter/findPasswordGetPhoneValidCode",
            "/sys/userCenter/findPasswordGetEmailValidCode",
            "/sys/userCenter/findPasswordByPhone",
            "/sys/userCenter/findPasswordByEmail"
    };
    /**
     * 仅超管使用的接口地址集合
     */
    private static final String[] SUPER_PERMISSION_PATH_ARR = {
            "/auth/session/**",
            "/auth/third/page",
            "/client/user/**",
            "/sys/org/**",
            "/sys/position/**",
            "/sys/button/**",
            "/sys/menu/**",
            "/sys/module/**",
            "/sys/role/**",
            "/sys/user/**",
    };

    /**
     * 注册 Sa-Token 路由拦截器
     */
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // 注册路由拦截器，自定义验证规则
        registry.addInterceptor(new SaInterceptor(handler -> {
                AllUrlHandler allUrlHandler = SpringUtils.getBean(AllUrlHandler.class);
                // 登录验证 -- 排除多个路径
                SaRouter
                    // 获取所有的
                    .match(allUrlHandler.getUrls())  // 拦截的 path 列表
                    .check(() -> {
                        // 检查是否登录 是否有token
                        StpUtil.checkLogin();

                        // 检查 header 与 param 里的 clientid 与 token 里的是否一致
                        //String headerCid = ServletUtils.getRequest().getHeader(LoginHelper.CLIENT_KEY);
                        //String paramCid = ServletUtils.getParameter(LoginHelper.CLIENT_KEY);
//                        String clientId = StpUtil.getExtra(LoginHelper.CLIENT_KEY).toString();
//                        if (!StringUtils.equalsAny(clientId, headerCid, paramCid)) {
//                            // token 无效
//                            throw NotLoginException.newInstance(StpUtil.getLoginType(),
//                                "-100", "客户端ID与Token不匹配",
//                                StpUtil.getTokenValue());
//                        }

                    });
            })).addPathPatterns("/**")
            // 排除不需要拦截的路径
            .excludePathPatterns(securityProperties.getExcludes());
    }

    /**
     * 注册跨域过滤器
     */
    //@Bean
    public SaServletFilter getSaServletFilter() {
        return new SaServletFilter()
                // 指定拦截路由
                .addInclude("/**").addExclude(NO_LOGIN_PATH_ARR)
                // 设置鉴权的接口
                .setAuth(r -> {
                    // B端的接口校验B端登录
                    SaRouter.match("/**")
                            // 排除无需登录接口
                            .notMatch(CollectionUtil.newArrayList(NO_LOGIN_PATH_ARR))
                            // 校验B端登录
                            .check(r1 -> StpUtil.checkLogin());
                    // B端的超管接口校验B端超管角色
//                    SaRouter.match("/**")
//                            // 排除无需登录接口
//                            .notMatch(CollectionUtil.newArrayList(NO_LOGIN_PATH_ARR))
//                            // 匹配超管接口
//                            .match(CollectionUtil.newArrayList(SUPER_PERMISSION_PATH_ARR))
//                            // 校验B端超管角色
//                            .check(r1 -> StpUtil.checkRole("superAdmin"));
                })

                // 前置函数：在每次认证函数之前执行
                .setBeforeAuth(obj -> {
                    // ---------- 设置跨域响应头 ----------
                    SaHolder.getResponse()
                            // 是否可以在iframe显示视图： DENY=不可以 | SAMEORIGIN=同域下可以 | ALLOW-FROM uri=指定域名下可以
                            // .setHeader("X-Frame-Options", "SAMEORIGIN")
                            // 是否启用浏览器默认XSS防护： 0=禁用 | 1=启用 | 1; mode=block 启用, 并在检查到XSS攻击时，停止渲染页面
                            .setHeader("X-XSS-Protection", "1; mode=block")
                            // 禁用浏览器内容嗅探
                            .setHeader("X-Content-Type-Options", "nosniff")
                            // 允许指定域访问跨域资源
                            .setHeader("Access-Control-Allow-Origin", "*")
                            // 允许所有请求方式
                            .setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE")
                            // 有效时间
                            .setHeader("Access-Control-Max-Age", "3600")
                            // 允许的header参数
                            .setHeader("Access-Control-Allow-Headers", "*");

                    // 如果是预检请求，则立即返回到前端
                    SaRouter.match(SaHttpMethod.OPTIONS)
                            // OPTIONS预检请求，不做处理
                            .free(r -> {})
                            .back();
                })

                // 异常处理
                .setError(e -> {
                    // 由于过滤器中抛出的异常不进入全局异常处理，所以必须提供[异常处理函数]来处理[认证函数]里抛出的异常
                    // 在[异常处理函数]里的返回值，将作为字符串输出到前端，此处统一转为JSON输出前端
                    SaResponse saResponse = SaHolder.getResponse();
                    saResponse.setHeader("Content-Type", "application/json;charset=" +  CharsetUtil.UTF_8);
                    return GlobalExceptionUtil.getCommonResult((Exception) e);
                });
    }
}
